import os
import numpy as np
import pandas as pd
from utils import plot_spectrum, Constants

if __name__ == '__main__':
    constants = Constants()
    material = "Polyethylene"
    labels = constants.labels
    filenames = constants.filenames
    augment_path = constants.augment_path
    img_path = constants.img_path
    img_filenames = constants.img_filenames
    augment_img_filename = os.path.join(img_path, img_filenames["augment"][material])
    augment_filename = os.path.join(augment_path, filenames["augment"][material])
    title = material
    # pd.read_csv("")
    print(labels)
    print(augment_filename)
    print(augment_img_filename)
    augment_data = pd.read_csv(augment_filename).values
    plot_spectrum(augment_data, augment_img_filename, title)
